export CUDA_VISIBLE_DEVICES=0
python main.py --base configs/autoencoder/autoencoder_kl_32x32x8_ft.yaml -t --gpus=0,